package org.coode.owl.mngr.impl;

import org.coode.owl.mngr.OWLServer;
import org.semanticweb.owl.model.*;

import java.net.URI;
/*
* Copyright (C) 2007, University of Manchester
*
* Modifications to the initial code base are copyright of their
* respective authors, or their employers as appropriate.  Authorship
* of the modifications may be determined from the ChangeLog placed at
* the end of this file.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.

* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
* Lesser General Public License for more details.

* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/

/**
 * Author: Nick Drummond<br>
 * http://www.cs.man.ac.uk/~drummond/<br><br>
 * <p/>
 * The University Of Manchester<br>
 * Bio Health Informatics Group<br>
 * Date: Aug 8, 2007<br><br>
 */
public class LabelShortFormProvider extends FragmentShortFormProvider {

    private OWLServer server;

    private URI uri;


    public LabelShortFormProvider(OWLServer server, URI annotationURI) {
        this.server = server;
        this.uri = annotationURI;
    }


    public String getShortForm(OWLEntity entity) {
        for (OWLOntology ont : server.getActiveOntologies()){
            for (OWLAnnotation annot : (entity).getAnnotations(ont)){
                final URI annotationURI = annot.getAnnotationURI();
                if (annotationURI.equals(uri)){
                    if (annot instanceof OWLConstantAnnotation){
                        return ((OWLConstantAnnotation)annot).getAnnotationValue().getLiteral();
                    }
                }
            }
        }
        return super.getShortForm(entity);
    }

    public String getShortForm(OWLNamedObject obj) {
        if (obj instanceof OWLOntology){
            return super.getShortForm(obj);
        }
        else{
            return getShortForm((OWLEntity)obj);
        }
    }
}